import os
import sys

# 数据集路径
if sys.platform.startswith('win'):
    DATASET_PATH = r'C:\Users\akitalee\Desktop\creditcard.csv'
elif sys.platform.startswith('linux'):
    DATASET_PATH = r'/home/sun/wzz/dataset/creditcard.csv'
elif sys.platform.startswith('darwin'):
    DATASET_PATH = r'备用'
else:
    DATASET_PATH = r'备用'

# 训练的批次
BATCH_SIZE = 16

# 训练的轮次
EPOCH = 500

# 设置数据加载的进程数目
NUM_WORKERS = 2

# 定位项目根目录的绝对路径
curPath = os.path.abspath(os.path.dirname(__file__))
PROJECT_PATH = curPath[:curPath.find('wzz-ae') + len('wzz-ae')]

# 日志保存路径
if not os.path.exists(os.path.join(PROJECT_PATH, 'log')):
    os.mkdir(path=os.path.join(PROJECT_PATH, 'log'))
LOG_PATH = os.path.join(PROJECT_PATH, 'log')
